import os
import random
import requests
import re
import urllib.request
import xlrd
import xlwt

#在path路径下创建一个test.xls文件，line行column列的每个单元格数据填充为1
def write_excel(path, sheet_name, line, column):
    file = path + "\\" + "test.xls"
    workbook = xlwt.Workbook()  # 新建一个工作簿
    sheet = workbook.add_sheet(sheet_name)  # 在工作簿中新建一个表格
    for i in range(0, line):  #行数
        for j in range(0, column):  #列数
            sheet.write(i, j, '1')  # 向表格中写入数据（对应的行和列）
    workbook.save(file)  # 保存工作簿
    print("xls格式表格写入数据成功！")

#在path路径下创建一个x页的test.doc文件
def write_word(path, x):
    file = path + "\\test.doc"
    f = open(file, 'w')
    for i in range(x):
        for i in range(44):
            f.write('test..........................\n')
    f.close()

#在目标路径下生成i个文件夹，命名为1—i
def dirs(i):
    path = 'C:\\Users\\Javid\\Desktop\\test1\\' #设置创建后文件夹存放的位置
    for i in range(1, i+1): #循环创建i次
       # 定义一个变量判断文件是否存在,path指代路径,str(i)指代文件夹的名字
        isExists = os.path.exists(path+str(i))
        if not isExists:                  #判断如果文件不存在,则创建
            os.makedirs(path+str(i))
            print("%s 目录创建成功"%i)
        else:
            print("%s 目录已经存在"%i)
            continue

#在目标路径下生成s个size兆的文件
def files(s, size):
    ss = size * 1024 * 1024
    # 生成文件
    for i in range(1, s + 1):
        filename = "C:\\Users\\Javid\\Desktop\\test1\\" + str(i) + ".txt"
        ds = 0
        f = open(filename, 'w')
        while ds < ss:   #size决定生成文件的大小
            f.write('测试文件'+str(round(random.uniform(-1000, 1000), 2)))
            f.write("\n")
            ds = os.path.getsize(filename)  #获取文件的字节大小

#在目标路径下生成x个文件夹，并在各个文件夹下分别生成s个size兆的文件
def files2(s, size, x):
    ss = size * 1024 * 1024
    path = 'C:\\Users\\Javid\\Desktop\\test1\\'
    for i in range(1, x+1):
        isExists = os.path.exists(path + str(i))
        if not isExists:
            os.makedirs(path + str(i))
            print("%s 目录创建成功" % i)
        for j in range(1, s+1):
            filename = "C:\\Users\\Javid\\Desktop\\test1\\"+str(i)+"\\" + str(j) + ".txt"
            ds = 0
            f = open(filename, 'w')
            while ds < ss:
                f.write('测试文件'+str(round(random.uniform(-1000, 1000), 2)))
                f.write("\n")
                ds = os.path.getsize(filename)


if __name__ == '__main__':
    #files2(1, 1, 1)
    #write_word('C:\\Users\\Javid\\Desktop\\test1',  2)